<template>
  <div>
    <vxe-gantt
      ref="ganttRef"
      v-bind="ganttOptions"
      @cell-click="cellClickEvent"
      @task-cell-click="taskCellClickEvent">
    </vxe-gantt>
  </div>
</template>

<script lang="ts">
import Vue from 'vue'
import { VxeUI } from 'vxe-pc-ui'
import { VxeGanttInstance, VxeGanttProps } from 'vxe-gantt'

interface RowVO {
  id: number
  title: string
  start: string
  end: string
  progress: number
}

export default Vue.extend({
  data () {
    const ganttOptions: VxeGanttProps<RowVO> = {
      height: 300,
      loading: false,
      rowConfig: {
        isCurrent: true,
        isHover: true
      },
      currentRowConfig: {
        trigger: 'manual'
      },
      taskBarConfig: {
        showProgress: true,
        showContent: true,
        barStyle: {
          round: true,
          bgColor: '#fca60b',
          completedBgColor: '#65c16f'
        }
      },
      taskViewConfig: {
        tableStyle: {
          width: 480
        }
      },
      columns: [
        { type: 'seq', width: 70 },
        { field: 'title', title: '任务名称' },
        { field: 'start', title: '开始时间', width: 100 },
        { field: 'end', title: '结束时间', width: 100 }
      ],
      data: [
        { id: 10001, title: 'A项目', start: '2024-03-01', end: '2024-03-04', progress: 3 },
        { id: 10002, title: '城市道路修理进度', start: '2024-03-03', end: '2024-03-08', progress: 10 },
        { id: 10003, title: 'B大工程', start: '2024-03-03', end: '2024-03-11', progress: 90 },
        { id: 10004, title: '超级大工程', start: '2024-03-05', end: '2024-03-11', progress: 15 },
        { id: 10005, title: '地球净化项目', start: '2024-03-08', end: '2024-03-15', progress: 100 },
        { id: 10006, title: '一个小目标项目', start: '2024-03-10', end: '2024-03-21', progress: 5 },
        { id: 10007, title: '某某计划', start: '2024-03-15', end: '2024-03-24', progress: 70 },
        { id: 10008, title: '某某科技项目', start: '2024-03-20', end: '2024-03-29', progress: 50 },
        { id: 10009, title: '地铁建设工程', start: '2024-03-19', end: '2024-03-20', progress: 5 },
        { id: 10010, title: '铁路修建计划', start: '2024-03-12', end: '2024-03-20', progress: 10 }
      ]
    }

    return {
      ganttOptions
    }
  },
  methods: {
    cellClickEvent ({ row }) {
      // 异步判断是否选中
      this.ganttOptions.loading = true
      setTimeout(() => {
        if (row.progress > 30) {
          const $gantt = this.$refs.ganttRef as VxeGanttInstance<RowVO>
          if ($gantt) {
            $gantt.setCurrentRow(row)
          }
        } else {
          VxeUI.modal.message({
            content: '禁止选中',
            status: 'error'
          })
        }
        this.ganttOptions.loading = false
      }, 300)
    },
    taskCellClickEvent ({ row }) {
      // 异步判断是否选中
      this.ganttOptions.loading = true
      setTimeout(() => {
        if (row.progress > 30) {
          const $gantt = this.$refs.ganttRef as VxeGanttInstance<RowVO>
          if ($gantt) {
            $gantt.setCurrentRow(row)
          }
        } else {
          VxeUI.modal.message({
            content: '禁止选中',
            status: 'error'
          })
        }
        this.ganttOptions.loading = false
      }, 300)
    }
  }
})
</script>
